#!/bin/bash

# test_config_path=/home/work_nfs7/xlgeng/data_handle_tts/pys/music_denoise/config/denoise.sh.yml
test_config_path=/home/work_nfs6/lhma/data_handle_tts/pys/music_denoise/config/test_dataset[tme_real_0]_model[sdprnn_stcm_pqmf_complexfull_simplesub_nocausal].yml
use_cuda=1
print_freq=10
save_freq=-1

save_case=$2
scp=$3
ts_pergpu=$4

# exp_dir=/home/work_nfs7/xlgeng/data_handle_tts/pys/music_denoise/exp/best
exp_dir=/home/work_nfs6/lhma/data_handle_tts/pys/music_denoise/exp/new
if [ ! -d ${exp_dir} ] ; then
    mkdir -p ${exp_dir}
fi

gpuid=$1

CUDA_VISIBLE_DEVICES=${gpuid} /home/environment/wmxu/anaconda3/envs/asteroid_alpha/bin/python -u /home/work_nfs6/lhma/data_handle_tts/pys/music_denoise/steps/atest_multithread.py \
    --config_path $test_config_path \
    --exp_dir $exp_dir \
    --save_freq $save_freq \
    --use_cuda $use_cuda \
    --output_dir $save_case \
    --scp $scp \
    --ts_pergpu $ts_pergpu

